import argparse
import json


_parser = argparse.ArgumentParser()
# 流地址
_parser.add_argument('--input', type=str, default='rtsp://admin:a123456789@192.168.1.65:554/Streaming/Channels/101?transportmode=unicast&profile=Profile_1', help="输入视频流（OpenCV参数）")
_parser.add_argument('--output', type=str, default='rtmp://localhost/live/1699323368155858390', help="输出视频流（FFMpeg参数）")
# 报警参数
_parser.add_argument('--alert_collect_url', type=str, help="报警的url", default='http://localhost:5000/alert/collect/1699323368155858390')
_parser.add_argument('--alert_sensitivity', type=float, help='报警的灵敏度，数值越高越灵敏', default=0.3)
_parser.add_argument('--alert', type=json.loads, nargs='+', help='报警事件，传入规定格式的Json字符串', default=None)
# 区域入侵参数
_parser.add_argument('--ai_region', type=json.loads, nargs='+', help='报警的区域，按照传入规定格式的Json字符串', default=None)
# 转播参数
_parser.add_argument('--resolution', type=str, help="转播的分辨率(宽*高)，如640*480", default=None)
_parser.add_argument('--cbr', type=int, help='转播的码率(单位为kbps)，如1024', default=1024)
_parser.add_argument('--fps', type=int, help='转播的帧率，如20', default=None)
# 画面参数
_parser.add_argument('--enable_draw_video_box', action='store_true', help="启用视频画面绘制识别框")
_parser.add_argument('--enable_draw_alert_box', action='store_true', help="启用报警图像绘制识别框")
# 多进程参数
_parser.add_argument('--enable_multiprocess', action='store_true', help="开启多进程检测，占用资源会更多，但画面会更流畅。如不开启，默认会使用单进程预测与补帧算法")
_parser.add_argument('--mutiprocess_worker_num', type=int, help="检测进程数量，如果设为1则与不开启多进程检测效果相同，默认为3", default=3)

config, _ = _parser.parse_known_args()
